#pragma once
#include "Common.hpp"
#include"InetAddr.hpp"
#include <iostream>
#include <cstring>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include<unistd.h>
#include<sys/wait.h>
#include<signal.h>
#include<pthread.h>
#include<functional>

static const uint16_t gport = 8888;
using handler_t =std::function<std::string(std::string str)>;
#define BACKLOG 8

class TcpServer
{
    struct ThreadData
    {
        int  sockfd;
        TcpServer *self;
    };
public:
    TcpServer(handler_t handler,uint16_t port = gport)
        : _port(port), _isruning(false),_handler(handler)
    {
    }
    void InitServer()
    {
        // 1.创建套接字
        _listensockfd = ::socket(AF_INET, SOCK_STREAM, 0); // TCP Socket创建好了
        if (_listensockfd < 0)
        {
            // 失败了
            std::cout << "TCP套接字创建失败！！" << std::endl;
            Die(SOCKET_ERR);
        }
        std::cout << "创建成功" << std::endl;
        sockaddr_in local;
        memset(&local, 0, sizeof(local));
        local.sin_family = AF_INET;
        local.sin_port = htons(_port);      // 结构体是要连通信息一起通过网络出去的，所以里面的port必须是网络序列的
        local.sin_addr.s_addr = INADDR_ANY; // 任意ip地址
        // 2.bind套接字
        int n = ::bind(_listensockfd, CONV(&local), sizeof(local));
        if (n < 0)
        {
            std::cout << "绑定失败" << std::endl;
            Die(BIND_ERR);
        }
        std::cout << "bind成功" << std::endl;

        // 3.cs模式，tcp是面向链接的，所以tcp随时等待被链接
        // tcp需要将socket设置为监听转态
        n = ::listen(_listensockfd, BACKLOG);
        if (n < 0)
        {
            std::cout << "监听失败" << std::endl;
            Die(LISTEN_ERR);
        }
        std::cout << "监听成功" << std::endl;
        // 比UDP多一步监听，因为将主动的等待链接变成被动的等待连接被动只需要去监听有人来就接受
        ::signal(SIGCHLD, SIG_IGN); // 不让父进程阻塞方法1：忽略子进程退出信号，防止僵尸进程
    }
    void HandlerRequest(int sockfd)//tcp也是全双工通信的
    {
        while(true)
        {
            //约定用户发过来的是一个完整的命令string，然后服务器返回一个string
            char inbuffer[1024];
            ssize_t n=read(sockfd,inbuffer,sizeof(inbuffer)-1);
            if(n>0)
            {
                inbuffer[n]=0;
                // std::string echo_str="server echo# ";
                // echo_str+=inbuffer;
                std::string cmd_result = _handler(inbuffer);
                ::write(sockfd,cmd_result.c_str(),cmd_result.size());
            }
            else if(n==0)
            {
                //read读到标识客户端退出了
                std::cout << "客户端退出" << std::endl;
                break;
            }
            else
            {
                //读取失败了；
                break;
            }
        }
        ::close(sockfd);
    }
    //为什么要加static是因为成员函数自带一个this指针，但是static成员函数没有this指针，不能访问非静态成员变量，我们直接定义一个struct结构体包含一个this指针通过这个去调用函数即可。
    static void *ThreadEntry(void *arg)
    {
        ThreadData* data=(ThreadData*)arg;
        data->self->HandlerRequest(data->sockfd);
        return nullptr;
    }
    void Start()
    {
        _isruning = true;
        while (_isruning)
        {
            // tcp一开始不能读取数据
            // 1.获取新链接
            sockaddr_in cli;
            socklen_t size = sizeof(cli);
            int sockfd = ::accept(_listensockfd, CONV(&cli), &size);//返回一个字节流。文件，管道都是字节流的，所以tcp也可以直接使用文件的相关的接口比如read，write。那么此时sockfd 就是文件描述符
            //我们要获取客户端的信息：数据+客户端的socket信息。 
            if(sockfd < 0)
            {
                std::cout << "accept失败" << std::endl;
                continue;
            }
            // 获取链接成功了
            std::cout << "accept成功" << std::endl;

            InetAddr addr(cli);
            std::cout << "客户端地址：" << addr.Addr() << ":" << addr.Port() << std::endl;
            // version-0,不用的只是举一个例子平常是不会这么用的。
            //HandlerRequest(sockfd);
            //vesrion-1,多进程版本,效率低写时拷贝，创建页表。
            // pid_t pid = fork();
            // if(pid==0)
            // {
            //     //子进程处理请求
            //     //子进程可以继承父进程的文件描述符表，父子各一张表，子进程可以操作父进程的文件描述符，但是不能操作自己，所以子进程要关闭父进程的套接字
            //     ::close(_listensockfd);
            //     //不让父进程阻塞方法2：除了忽略子进程退出信号还可以
            //     if(fork()>0) exit(0);//子进程退出
            //     //这是孙子进程，父亲不在了只有爷爷，但是爷爷没有管理权就变成了孤儿进程
            //     //孙子进程->孤儿进程
            //     //所以爷爷进程直接wait成功，继续进行代码就不会阻塞了。
            //     //孤儿进程会被OS领养自动回收。
            //     HandlerRequest(sockfd);
            //     exit(0);
            // }
            //     ::close(sockfd);
                //要关闭不需要的套接字，防止资源泄露
                //关闭文件本质是引用计数--
                //这会阻塞等待就不会再accept了，但是不等待就会有僵尸进程，所以这里直接signal忽略子进程的退出信号
                // int rid=::waitpid(pid,nullptr,0);
                // if(rid<0)
                // {
                //     std::cout<<"waitpid失败"<<std::endl;
                // }
            //vesion-2,多线程版本
            //主线程和新线程是共享一张文件描述符表的！！！
            //所以就不需要关闭不要的文件描述符
            pthread_t tid;
            ThreadData *data=new ThreadData;
            data->sockfd=sockfd;
            data->self=this;
            //假设这次直接传入sockfd的话，如果新线程没有跑起来或者没有跑完，主线程就又accept了一个新的sockfd那么就乱了，所以这里得先new一个出来
            pthread_create(&tid,nullptr,ThreadEntry,&(data->sockfd));
            //vesion-3,线程池版本
            //TODO
            //远程命令实现

        }
        _isruning = false;
    }
    ~TcpServer()
    {
    }

private:
    int _listensockfd;
    uint16_t _port;
    bool _isruning;
    handler_t _handler;

    //处理上层命令的入口
};